home *** CD-ROM | disk | FTP | other *** search
/ Mac Easy 2010 May / Mac Life Ubuntu.iso / casper / filesystem.squashfs / usr / share / apport / testsuite / test-hooks < prev   
Encoding:
Text File  |  2009-04-06  |  9.1 KB  |  247 lines
  1. #!/usr/bin/python
  2. '''the various package hooks.'''
  3.  
  4. import unittest, subprocess, tempfile, os, shutil, os.path, sys, optparse
  5. from cStringIO import StringIO
  6. import apport, apport.fileutils
  7.  
  8. # parse command line options
  9. optparser = optparse.OptionParser('%prog [options]')
  10. optparser.add_option('-l', '--local',
  11.     help='Test scripts in ./bin/ instead of /usr/share/apport/',
  12.     action='store_true', dest='local', default=False)
  13. options, args = optparser.parse_args()
  14.  
  15. if options.local:
  16.     basedir = 'bin'
  17. else:
  18.     basedir = '/usr/share/apport'
  19.  
  20. class _HooksTest(unittest.TestCase):
  21.     def setUp(self):
  22.         self.orig_report_dir = apport.fileutils.report_dir
  23.         apport.fileutils.report_dir = tempfile.mkdtemp()
  24.         os.environ['APPORT_REPORT_DIR'] = apport.fileutils.report_dir
  25.  
  26.         self.workdir = tempfile.mkdtemp()
  27.  
  28.     def tearDown(self):
  29.         shutil.rmtree(apport.fileutils.report_dir)
  30.         apport.fileutils.report_dir = self.orig_report_dir 
  31.  
  32.         shutil.rmtree(self.workdir)
  33.  
  34.     def test_package_hook_nologs(self):
  35.         '''package_hook without any log files.'''
  36.  
  37.         ph = subprocess.Popen(['%s/package_hook' % basedir, '-p', 'bash'],
  38.             stdin=subprocess.PIPE)
  39.         ph.communicate('something is wrong')
  40.         self.assertEqual(ph.returncode, 0, 'package_hook finished successfully')
  41.  
  42.         reps = apport.fileutils.get_new_reports()
  43.         self.assertEqual(len(reps), 1, 'package_hook created a report')
  44.  
  45.         r = apport.Report()
  46.         r.load(open(reps[0]))
  47.  
  48.         self.assertEqual(r['ProblemType'], 'Package')
  49.         self.assertEqual(r['Package'], 'bash')
  50.         self.assertEqual(r['SourcePackage'], 'bash')
  51.         self.assertEqual(r['ErrorMessage'], 'something is wrong')
  52.  
  53.     def test_package_hook_uninstalled(self):
  54.         '''package_hook on an uninstalled package (might fail to install).'''
  55.  
  56.         #FIXME: This is a really Debian/Ubuntu specific dodgy test case
  57.         ph = subprocess.Popen(['%s/package_hook' % basedir, '-p', 'libdb4.7-tcl'],
  58.             stdin=subprocess.PIPE)
  59.         ph.communicate('something is wrong')
  60.         self.assertEqual(ph.returncode, 0, 'package_hook finished successfully')
  61.  
  62.         reps = apport.fileutils.get_new_reports()
  63.         self.assertEqual(len(reps), 1, 'package_hook created a report')
  64.  
  65.         r = apport.Report()
  66.         r.load(open(reps[0]))
  67.  
  68.         self.assertEqual(r['ProblemType'], 'Package')
  69.         self.assertEqual(r['Package'], 'libdb4.7-tcl')
  70.         self.assertEqual(r['SourcePackage'], 'db')
  71.         self.assertEqual(r['ErrorMessage'], 'something is wrong')
  72.  
  73.     def test_package_hook_logs(self):
  74.         '''package_hook with a log dir and a log file.'''
  75.  
  76.         open(os.path.join(self.workdir, 'log_1.log'), 'w').write('Log 1\nbla')
  77.         open(os.path.join(self.workdir, 'log2'), 'w').write('Yet\nanother\nlog')
  78.  
  79.         ph = subprocess.Popen(['%s/package_hook' % basedir, '-p', 'bash', '-l',
  80.             os.path.realpath(sys.argv[0]), '-l', self.workdir],
  81.             stdin=subprocess.PIPE)
  82.         ph.communicate('something is wrong')
  83.         self.assertEqual(ph.returncode, 0, 'package_hook finished successfully')
  84.  
  85.         reps = apport.fileutils.get_new_reports()
  86.         self.assertEqual(len(reps), 1, 'package_hook created a report')
  87.  
  88.         r = apport.Report()
  89.         r.load(open(reps[0]))
  90.  
  91.         filekey = None
  92.         log1key = None
  93.         log2key = None
  94.         for k in r.keys():
  95.             if k.endswith('Testhooks'):
  96.                 filekey = k
  97.             elif k.endswith('Log1log'):
  98.                 log1key = k
  99.             elif k.endswith('Log2'):
  100.                 log2key = k
  101.  
  102.         self.assert_(filekey)
  103.         self.assert_(log1key)
  104.         self.assert_(log2key)
  105.         self.assert_('0234lkjas' in r[filekey])
  106.         self.assertEqual(len(r[filekey]), os.path.getsize(sys.argv[0]))
  107.         self.assertEqual(r[log1key], 'Log 1\nbla')
  108.         self.assertEqual(r[log2key], 'Yet\nanother\nlog')
  109.  
  110.     def test_kernel_crashdump(self):
  111.         '''kernel_crashdump.'''
  112.  
  113.         f = open(os.path.join(apport.fileutils.report_dir, 'vmcore'), 'w')
  114.         f.write('\x01' * 100)
  115.         f.close()
  116.         f = open(os.path.join(apport.fileutils.report_dir, 'vmcore.log'), 'w')
  117.         f.write('vmcore successfully dumped')
  118.         f.close()
  119.  
  120.         self.assertEqual(subprocess.call('%s/kernel_crashdump' % basedir), 0,
  121.             'kernel_crashdump finished successfully')
  122.  
  123.         reps = apport.fileutils.get_new_reports()
  124.         self.assertEqual(len(reps), 1, 'kernel_crashdump created a report')
  125.  
  126.         r = apport.Report()
  127.         r.load(open(reps[0]))
  128.  
  129.         self.assertEqual(set(r.keys()), set(['Date', 'SourcePackage',
  130.             'Package', 'ProblemType', 'VmCore', 'VmCoreLog']))
  131.         self.assertEqual(r['ProblemType'], 'KernelCrash')
  132.         self.assertEqual(r['VmCoreLog'], 'vmcore successfully dumped')
  133.         self.assertEqual(r['VmCore'], '\x01' * 100)
  134.         self.assertEqual(r['SourcePackage'], 'linux')
  135.  
  136.         self.assert_(os.uname()[2].split('-')[0] in r['Package'])
  137.  
  138.         r.add_package_info(r['Package'])
  139.         self.assert_(' ' in r['Package']) # appended version number
  140.         r.add_hooks_info()
  141.         self.assert_('ProcModules' in r)
  142.         self.assert_('Lspci' in r)
  143.  
  144.     @classmethod
  145.     def _gcc_version_path(klass):
  146.         '''Determine a valid version and executable path of gcc and return it
  147.         as a tuple.'''
  148.  
  149.         gcc = subprocess.Popen(['gcc', '--version'], stdout=subprocess.PIPE)
  150.         out = gcc.communicate()[0]
  151.         assert gcc.returncode == 0, '"gcc --version" must work for this test suite'
  152.  
  153.         gcc_ver = '.'.join(out.splitlines()[0].split()[2].split('.')[:2])
  154.         gcc_path = '/usr/bin/gcc-' + gcc_ver
  155.  
  156.         assert subprocess.call([gcc_path, '--version'], stdout=subprocess.PIPE) == 0, \
  157.             gcc_path + ' must exist and work for this test suite'
  158.  
  159.         return (gcc_ver, gcc_path)
  160.  
  161.     def test_gcc_ide_hook_file(self):
  162.         '''gcc_ice_hook with a temporary file.'''
  163.  
  164.         (gcc_version, gcc_path) = self._gcc_version_path()
  165.  
  166.         test_source = tempfile.NamedTemporaryFile()
  167.         test_source.write('int f(int x);')
  168.         test_source.flush()
  169.         test_source.seek(0)
  170.  
  171.         self.assertEqual(subprocess.call(['%s/gcc_ice_hook' % basedir,
  172.             gcc_path, test_source.name]), 0, 'gcc_ice_hook finished successfully')
  173.  
  174.         reps = apport.fileutils.get_new_reports()
  175.         self.assertEqual(len(reps), 1, 'gcc_ice_hook created a report')
  176.  
  177.         r = apport.Report()
  178.         r.load(open(reps[0]))
  179.  
  180.         self.assertEqual(r['ProblemType'], 'Crash')
  181.         self.assertEqual(r['ExecutablePath'], gcc_path)
  182.         self.assertEqual(r['PreprocessedSource'], test_source.read())
  183.  
  184.         r.add_package_info()
  185.  
  186.         self.assert_(r['Package'].startswith('gcc-' + gcc_version))
  187.         self.assert_(r['SourcePackage'].startswith('gcc'))
  188.  
  189.     def test_gcc_ide_hook_pipe(self):
  190.         '''gcc_ice_hook with piping.'''
  191.  
  192.         (gcc_version, gcc_path) = self._gcc_version_path()
  193.  
  194.         test_source = 'int f(int x);'
  195.  
  196.         hook = subprocess.Popen(['%s/gcc_ice_hook' % basedir, gcc_path, '-'], 
  197.             stdin=subprocess.PIPE)
  198.         hook.communicate(test_source)
  199.         self.assertEqual(hook.returncode, 0, 'gcc_ice_hook finished successfully')
  200.  
  201.         reps = apport.fileutils.get_new_reports()
  202.         self.assertEqual(len(reps), 1, 'gcc_ice_hook created a report')
  203.  
  204.         r = apport.Report()
  205.         r.load(open(reps[0]))
  206.  
  207.         self.assertEqual(r['ProblemType'], 'Crash')
  208.         self.assertEqual(r['ExecutablePath'], gcc_path)
  209.         self.assertEqual(r['PreprocessedSource'], test_source)
  210.  
  211.         r.add_package_info()
  212.  
  213.         self.assert_(r['Package'].startswith('gcc-' + gcc_version))
  214.         self.assert_(r['SourcePackage'].startswith('gcc'))
  215.  
  216.     def test_kernel_oops_hook(self):
  217.     test_source = '''------------[ cut here ]------------
  218. kernel BUG at /tmp/oops.c:5!
  219. invalid opcode: 0000 [#1] SMP 
  220. Modules linked in: oops cpufreq_stats ext2 i915 drm nf_conntrack_ipv4 ipt_REJECT iptable_filter ip_tables nf_conntrack_ipv6 xt_state nf_conntrack xt_tcpudp ip6t_ipv6header ip6t_REJECT ip6table_filter ip6_tables x_tables ipv6 loop dm_multipath rtc_cmos iTCO_wdt iTCO_vendor_support pcspkr i2c_i801 i2c_core battery video ac output power_supply button sg joydev usb_storage dm_snapshot dm_zero dm_mirror dm_mod ahci pata_acpi ata_generic ata_piix libata sd_mod scsi_mod ext3 jbd mbcache uhci_hcd ohci_hcd ehci_hcd
  221. '''
  222.         hook = subprocess.Popen(['%s/kernel_oops' % basedir],
  223.             stdin=subprocess.PIPE)
  224.         hook.communicate(test_source)
  225.         self.assertEqual(hook.returncode, 0, 'kernel_oops finished successfully')
  226.  
  227.         reps = apport.fileutils.get_new_reports()
  228.         self.assertEqual(len(reps), 1, 'kernel_oops created a report')
  229.  
  230.         r = apport.Report()
  231.         r.load(open(reps[0]))
  232.  
  233.         self.assertEqual(r['ProblemType'], 'KernelOops')
  234.         self.assertEqual(r['OopsText'], test_source)
  235.  
  236.         r.add_package_info(r['Package'])
  237.  
  238.         self.assert_(r['Package'].startswith('linux-image-'))
  239.  
  240. # call tests
  241.  
  242. tl = unittest.TestLoader()
  243. tests_all = unittest.TestSuite((
  244.     tl.loadTestsFromName('__main__')
  245. ))
  246. unittest.TextTestRunner(verbosity=2).run(tests_all)
  247.